tensor([[0.7600, 0.7500, 0.7600, 0.7500, 0.7600],
        [0.2400, 0.2500, 0.2400, 0.2500, 0.2400]], grad_fn=<RoundBackward1>)